from discriminator import Discriminator
from generator import Generator
import torch



if __name__ == '__main__':
    g = Generator()
    d = Discriminator()

    z = torch.randn(size=[4,100,1,1])
    img = g(z)
    print(img.shape)
    y = d(img)
    print(y.shape)